14. Pandas Query

我们要使用的另一个有用的函数是 Pandas 的 query 函数。

在上一课中,我们通过使用掩码进行索引来在数据框(DataFrame)中选择行。这些是同样的例子,以及使用 query() 的等效语句。

# selecting malignant records in cancer data
df_m = df[df['diagnosis'] == 'M']
df_m = df.query('diagnosis == "M"')

# selecting records of people making over $50K
df_a = df[df['income'] == ' >50K']
df_a = df.query('income == " >50K"')